/*
 * 项   目：code-generator
 * 作   者：LuoYiMin
 * 邮   箱：1725990142@qq.com
 * 创建时间：2018/12/27 15:33
 */
package com.roin.util.code.generator.util;

import freemarker.cache.NullCacheStorage;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;

public class FreemarkerManager extends Configuration{
    private static final Logger LOGGER = LoggerFactory.getLogger(FreemarkerManager.class);
    private static class SingletonHolder{
        private final static FreemarkerManager instance=new FreemarkerManager("/ftl");
    }
    public static FreemarkerManager getInstance(){
        return SingletonHolder.instance;
    }

    private FreemarkerManager(String templatePath){
        this.setClassForTemplateLoading(this.getClass(), templatePath);
        this.setDateTimeFormat("yyyy-MM-dd HH:mm:ss");
        this.setDateFormat("yyyy-MM-dd");
        this.setTimeFormat("HH:mm:ss");
        this.setCacheStorage(new NullCacheStorage());//不使用缓存
    }

    /**
     * 控制台输出
     *
     * @param name
     * @param root
     */
    public static void print(String name,Object root) {
        try {
            // 通过Template可以将模板文件输出到相应的流
            Template temp = getInstance().getTemplate(name);
            temp.process(root, new PrintWriter(System.out));
        } catch (TemplateException e) {
            LOGGER.error("模板渲染异常",e);
        } catch (IOException e) {
            LOGGER.error("模板渲染异常",e);
        }
    }

    /**
     * 输出HTML文件
     *
     * @param name
     * @param root
     * @param outFile
     */
    public static void outToFile(String name, Object root, String outFile) {
        FileWriter out = null;
        try {
            // 通过一个文件输出流，就可以写到相应的文件中，此处用的是绝对路径
            out = new FileWriter(new File(outFile));
            Template temp = getInstance().getTemplate(name);
            temp.process(root, out);
        } catch (IOException e) {
            LOGGER.error("模板渲染异常",e);
        } catch (TemplateException e) {
            LOGGER.error("模板渲染异常",e);
        } finally {
            try {
                if (out != null)
                    out.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}
